package com.liguangting.agentsflex.constant;

import java.util.Optional;


/**
 * @author liguangting
 * @date 2024/7/2
 * @description
 */
public class ApiKeys {

    // You can use "demo" api key for demonstration purposes.
    // You can get your own OpenAI API key here: https://platform.openai.com/account/api-keys
    public static final String OPENAI_API_KEY = Optional.ofNullable(System.getenv("OPENAI_API_KEY")).orElse("demo");

    // Spark LLM model
    public static final String SPARK_APP_ID = System.getenv("SPARK_APP_ID");
    public static final String SPARK_API_SECRET = System.getenv("SPARK_API_SECRET");
    public static final String SPARK_API_KEY = System.getenv("SPARK_API_KEY");

    // Qwen LLM model
    public static final String QWEN_API_KEY = System.getenv("QWEN_API_KEY");

    // ali vector db
    public static final String ALI_VECTOR_DB_API_KEY = System.getenv("ALI_VECTOR_DB_API_KEY");
    public static final String ALI_VECTOR_DB_ENDPOINT = System.getenv("ALI_VECTOR_DB_ENDPOINT");
    public static final String ALI_VECTOR_DB_COLLECTION = "test_docs";
}
